from .TAL_pytorch import TAL_Linear, TAL_Conv2d